package com.example.framework.interceptor;

import com.revinate.guava.util.concurrent.RateLimiter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * @Author 王鹤蒲
 * @Date 2024 02 29
 */

@Configuration
public class RateLimiterInterceptor implements HandlerInterceptor {

    private static final Logger logger= LoggerFactory.getLogger(RateLimiterInterceptor.class);

    private static final RateLimiter rateLimiter=RateLimiter.create(50);

   @Override
   public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        logger.info("阻塞中......");
        rateLimiter.acquire();
        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {

    }

}
